package com.example.java_chatroom.api;

import com.example.java_chatroom.mapper.UserMapper;
import com.example.java_chatroom.model.User;
import jakarta.annotation.Resource;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpSession;
import lombok.extern.slf4j.Slf4j;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;

@RestController
@Slf4j
public class UserAPI {
    @Resource
    UserMapper userMapper;

    @PostMapping("/login")
    @ResponseBody
    public Object login(String username, String password, HttpServletRequest req) {
        log.info("==========用户尝试登录,username:{}, password:{}============", username, password);
        // 1.先去数据库中查, 看 username 能否找到对应的 user 对象
        // 如果能匹配, 则看密码是否匹配
        User user = userMapper.selectByName(username);
        if(user == null || !user.getPassword().equals(password)) {
            //登录失败 返回一个空的对象
            log.info("登录失败, 用户名或密码错误!" + user);
            return new User();
        }

        // 2.如果能匹配, 登录成功, 创建会话
        HttpSession session = req.getSession(true);
        session.setAttribute("user", user);
        // 在返回之前, 把password删除
        user.setPassword("");
        return user;
    }

    @PostMapping("/register")
    @ResponseBody
    public Object register(String username, String password) {
        log.info("=========用户注册,username:{}, password:{}============", username, password);
        User user = null;
        try {
            user = new User();
            user.setUsername(username);
            user.setPassword(password);
            int ret = userMapper.insert(user);
            log.info("注册 ret:" + ret);
            user.setPassword("");
            log.info("注册成功!");
            return user;
        }catch (DuplicateKeyException e) {
            //如果 insert 方法抛出上述异常, 说明名字重复了, 注册失败
            e.printStackTrace();
            user = new User();
            log.info("注册失败, username="+username);
        }
        return user;
    }

    @GetMapping("/userInfo")
    @ResponseBody
    public Object getUserInfo(HttpServletRequest req) {
        //1.先从请求中获取会话
        HttpSession session = req.getSession(false);
        if(session == null) {
            //会话不存在, 用户尚未登录, 此时返回一个空的对象
            log.info("[getUserInfo] 当前获取不到 session 对象");
            return new User();
        }
        //2.从会话中获取到之前保存的用户对象
        User user = (User) session.getAttribute("user");
        if(user == null) {
            log.info("[getUserInfo] 当前获取不到 user 对象");
            return new User();
        }
        user.setPassword("");
        return user;
    }
}
